find a subset of set in python